package at.guigu.domain.vo;

import at.guigu.domain.po.User;
import cn.hutool.core.bean.BeanUtil;
import cn.hutool.core.collection.CollUtil;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import lombok.AllArgsConstructor;
import lombok.Data;
import lombok.NoArgsConstructor;

import java.util.Collections;
import java.util.List;
import java.util.function.Function;
import java.util.stream.Collectors;

@Data
@AllArgsConstructor(staticName = "of")
@NoArgsConstructor
public class PageVo<T> {
    private  Long total;
    private  Long Pages;
    private List<T> list;

    public <P> PageVo(Page<P> page, Class<T> clazz){
        // 总数据量
        this.total = page.getTotal();
        // 总页数
        this.Pages = page.getPages();
        // 当前页数据集合
        List<P> users = page.getRecords();
        // 若当前页无数据则返回空数据
        if (CollUtil.isEmpty(users)){
            list = Collections.emptyList();
            return;
        }
        // 当前页有数据则 将集合List<P>转换为集合List<T>
        this.list = BeanUtil.copyToList(users, clazz);
    }

    public <P> PageVo(Page<P> page, Function<P, T> convertor){
        // 总数据量
        this.total = page.getTotal();
        // 总页数
        this.Pages = page.getPages();
        // 当前页数据集合
        List<P> users = page.getRecords();
        // 若当前页无数据则返回空数据
        if (CollUtil.isEmpty(users)){
            list = Collections.emptyList();
            return;
        }
        // 当前页有数据则 将集合List<P>转换为集合List<T>
        this.list = users.stream().map(convertor).collect(Collectors.toList());
    }
}
